#!/bin/bash  
  
# 用途：设置GraphRag与Ollama搭配使用的环境  
# 系统：Ubuntu 22.04  
# 作者：AMD FAE Team huang sheng  
# 版本：测试版  
  
# 克隆GraphRag的GitHub仓库  
git clone https://github.com/TheAiSingularity/graphrag-local-ollama.git  
cd graphrag-local-ollama/  
  
# 创建&激活虚拟环境   
python3 -m venv amdgpu  
source amdgpu/bin/activate  
  
# 安装GraphRag  
pip install -e . 
pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.0 
  
# 以下是额外的设置步骤，已添加中文备注  
  
# 创建一个用于存储实验数据和结果的目录  
sudo mkdir -p ./ragtest/input  
sudo chown -R $(whoami):$(whoami) ~/ragtest/input

# 将示例数据复制到ragtest目录的input子目录中  
# cp input/* ./ragtest/input  
  
# 初始化ragtest文件夹，创建所需的文件  
python3 -m graphrag.index --init --root ./ragtest

